#ifndef UNBALANCED_ADABOOST_H__CMU_
#define UNBALANCED_ADABOOST_H__CMU_

#include "adaboost.h"
#include "real_adaboost.h"

class unbalanced_adaboost : public adaboost {
protected:
	double beta;
public:
	virtual void train_initialization(int & argc, char **& argv);
	virtual void initialize_distribution(double * d, double * y, int l);
};

class unbalanced_real_adaboost : public real_adaboost {
protected:
	double beta;
public:
	virtual void train_initialization(int & argc, char **& argv);
	virtual void initialize_distribution(double * d, double * y, int l);
};

#endif